package com.sesame.common.utils;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.geom.AffineTransform;
import java.awt.image.AffineTransformOp;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;

/**
 * Created by nealmiao on 12/30/2016.
 */
public class ImageUtils {

    /**
     * 图片格式：JPG
     */
    private static final String PICTRUE_FORMATE_JPG = "jpg";

    private ImageUtils() {
    }

    /**
     * 添加图片水印
     *
     * @param targetImg
     *         目标图片路径，如：C://myPictrue//1.jpg
     * @param waterImg
     *         水印图片路径，如：C://myPictrue//logo.png
     * @param x
     *         水印图片距离目标图片左侧的偏移量，如果x<0, 则在正中间
     * @param y
     *         水印图片距离目标图片上侧的偏移量，如果y<0, 则在正中间
     * @param alpha
     *         透明度(0.0 -- 1.0, 0.0为完全透明，1.0为完全不透明)
     */
    public final static void pressImage(String targetImg, String waterImg, int x, int y, float alpha, Integer mulFileServer) {
        pressImage(targetImg, waterImg, x, y, alpha, null,mulFileServer);
    }

    /**
     * 添加图片水印，默认右下角
     *
     * @param targetImg
     *         目标图片路径，如：C://myPictrue//1.jpg
     * @param waterImg
     *         水印图片路径，如：C://myPictrue//logo.png
     * @param x
     *         水印图片距离目标图片左侧的偏移量，如果x<0, 则在正中间
     * @param y
     *         水印图片距离目标图片上侧的偏移量，如果y<0, 则在正中间
     * @param alpha
     *         透明度(0.0 -- 1.0, 0.0为完全透明，1.0为完全不透明)
     * @param fileNameAppend
     *         转换后文件名称后缀，null或者空表示不改名字
     */
    public final static String pressImage(String targetImg, String waterImg, int x, int y, float alpha, String fileNameAppend,Integer mulFileServer) {
        try {
            File file = new File(targetImg);
            Image image = ImageIO.read(file);
            int width = image.getWidth(null);
            int height = image.getHeight(null);

            if (width < 400 || height < 400){
                return file.getName();
            }

            BufferedImage bufferedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
            Graphics2D g = bufferedImage.createGraphics();
            g.drawImage(image, 0, 0, width, height, null);

            Image waterImage = ImageIO.read(new File(waterImg));    // 水印文件
            int width_1 = waterImage.getWidth(null);
            int height_1 = waterImage.getHeight(null);
            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));

            int widthDiff = width - width_1;
            int heightDiff = height - height_1;
            if (x < 0) {
                x = widthDiff / 2;
            } else if (x > widthDiff) {
                x = widthDiff;
            }
            if (y < 0) {
                y = heightDiff / 2;
            } else if (y > heightDiff) {
                y = heightDiff;
            }
            g.drawImage(waterImage, width - x - width_1, height - y - height_1, width_1, height_1, null); // 水印文件结束
            g.dispose();
            String[] tmpPath = targetImg.split("\\.");
            File fileOut = new File(tmpPath[0] + (null == fileNameAppend ? "" : "_" + fileNameAppend) + "." + tmpPath[1]);
            ImageIO.write(bufferedImage, PICTRUE_FORMATE_JPG, fileOut);

            /*try {
                if(mulFileServer == 1){
                    SFTPUtil.uploadFile1(fileOut,DateUtils.getShortDateStr(),fileOut.getName());
                }
            }catch (Exception e){
                e.printStackTrace();
            }*/

            return fileOut.getName();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return "";
    }

    /**
     * 添加文字水印
     *
     * @param targetImg
     *         目标图片路径，如：C://myPictrue//1.jpg
     * @param pressText
     *         水印文字， 如：xxxx水印
     * @param fontName
     *         字体名称，	如：宋体
     * @param fontStyle
     *         字体样式，如：粗体和斜体(Font.BOLD|Font.ITALIC)
     * @param fontSize
     *         字体大小，单位为像素
     * @param color
     *         字体颜色
     * @param x
     *         水印文字距离目标图片左侧的偏移量，如果x<0, 则在正中间
     * @param y
     *         水印文字距离目标图片上侧的偏移量，如果y<0, 则在正中间
     * @param alpha
     *         透明度(0.0 -- 1.0, 0.0为完全透明，1.0为完全不透明)
     */
    public static void pressText(String targetImg, String pressText, String fontName, int fontStyle, int fontSize, Color color, int x, int y, float alpha) {
        pressText(targetImg, pressText, fontName, fontStyle, fontSize, color, x, y, alpha, null);
    }

    /**
     * 添加文字水印
     *
     * @param targetImg
     *         目标图片路径，如：C://myPictrue//1.jpg
     * @param pressText
     *         水印文字， 如：xxxx水印
     * @param fontName
     *         字体名称，	如：宋体
     * @param fontStyle
     *         字体样式，如：粗体和斜体(Font.BOLD|Font.ITALIC)
     * @param fontSize
     *         字体大小，单位为像素
     * @param color
     *         字体颜色
     * @param x
     *         水印文字距离目标图片左侧的偏移量，如果x<0, 则在正中间
     * @param y
     *         水印文字距离目标图片上侧的偏移量，如果y<0, 则在正中间
     * @param alpha
     *         透明度(0.0 -- 1.0, 0.0为完全透明，1.0为完全不透明)
     * @param fileNameAppend
     *         转换后文件名称后缀，null或者空表示不改名字
     */
    public static void pressText(String targetImg, String pressText, String fontName, int fontStyle, int fontSize, Color color, int x, int y, float alpha, String fileNameAppend) {
        try {
            File file = new File(targetImg);
            Image image = ImageIO.read(file);
            int width = image.getWidth(null);
            int height = image.getHeight(null);
            BufferedImage bufferedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
            Graphics2D g = bufferedImage.createGraphics();
            g.drawImage(image, 0, 0, width, height, null);
            g.setFont(new Font(fontName, fontStyle, fontSize));
            g.setColor(color);
            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));

            int width_1 = fontSize * getLength(pressText);
            int height_1 = fontSize;
            int widthDiff = width - width_1;
            int heightDiff = height - height_1;
            if (x < 0) {
                x = widthDiff / 2;
            } else if (x > widthDiff) {
                x = widthDiff;
            }
            if (y < 0) {
                y = heightDiff / 2;
            } else if (y > heightDiff) {
                y = heightDiff;
            }

            g.drawString(pressText, x, y + height_1);
            g.dispose();
            String[] tmpPath = targetImg.split("\\.");
            File fileOut = new File(tmpPath[0] + (null == fileNameAppend ? "" : "_" + fileNameAppend) + "." + tmpPath[1]);
            ImageIO.write(bufferedImage, PICTRUE_FORMATE_JPG, fileOut);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取字符长度，一个汉字作为 1 个字符, 一个英文字母作为 0.5 个字符
     *
     * @param text
     *
     * @return 字符长度，如：text="中国",返回 2；text="test",返回 2；text="中国ABC",返回 4.
     */
    public static int getLength(String text) {
        int textLength = text.length();
        int length = textLength;
        for (int i = 0; i < textLength; i++) {
            if (String.valueOf(text.charAt(i)).getBytes().length > 1) {
                length++;
            }
        }
        return (length % 2 == 0) ? length / 2 : length / 2 + 1;
    }

    /**
     * 图片缩放
     *
     * @param filePath
     *         图片路径
     * @param height
     *         高度
     * @param width
     *         宽度
     * @param bb
     *         比例不对时是否需要补白
     */
    public static void resize(String filePath, int height, int width, boolean bb,Integer mulFileServer) {
        resize(filePath, height, width, bb, null,mulFileServer);
    }

    /**
     * 图片缩放
     *
     * @param filePath
     *         图片路径
     * @param height
     *         高度
     * @param width
     *         宽度
     * @param bb
     *         比例不对时是否需要补白
     * @param fileNameAppend
     *         转换后文件名称后缀，null或者空表示不改名字
     */
    public static String resize(String filePath, int height, int width, boolean bb, String fileNameAppend,Integer mulFileServer) {
        try {
            double ratio = 0; //缩放比例
            File f = new File(filePath);

            BufferedImage bi = null;
            try {
                bi = ImageIO.read(f);
            } catch (IOException e) {
                e.printStackTrace();
            }



            if (bi.getWidth() < 400 || bi.getHeight() < 400){
                return filePath;
            }

            int dbWidth = bi.getWidth();
            int dbHeight = bi.getHeight();

            double bili = 0.0f;

            Double biliWidth = bi.getWidth() * 1.0 / width;
            Double biliHeight = bi.getHeight() * 1.0 / height;

            bili = biliWidth>biliHeight?biliWidth:biliHeight;

            if (bili < 0 ){
                return filePath;
            }

            height = (int)(bi.getHeight() / bili);
            width =  (int)(bi.getWidth() / bili);


            Image itemp = bi.getScaledInstance(width, height, BufferedImage.SCALE_SMOOTH);
            //计算比例
            if ((bi.getHeight() > height) || (bi.getWidth() > width)) {
                if (bi.getHeight() > bi.getWidth()) {
                    ratio = (new Integer(height)).doubleValue() / bi.getHeight();
                } else {
                    ratio = (new Integer(width)).doubleValue() / bi.getWidth();
                }
                AffineTransformOp op = new AffineTransformOp(AffineTransform.getScaleInstance(ratio, ratio), null);
                itemp = op.filter(bi, null);
            }
            if (bb) {
                BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
                Graphics2D g = image.createGraphics();
                g.setColor(Color.white);
                g.fillRect(0, 0, width, height);
                if (width == itemp.getWidth(null))
                    g.drawImage(itemp, 0, (height - itemp.getHeight(null)) / 2, itemp.getWidth(null), itemp.getHeight(null), Color.white, null);
                else
                    g.drawImage(itemp, (width - itemp.getWidth(null)) / 2, 0, itemp.getWidth(null), itemp.getHeight(null), Color.white, null);
                g.dispose();
                itemp = image;
            }
            String[] tmpPath = filePath.split("\\.");
            File fileOut = new File(tmpPath[0] + (null == fileNameAppend ? "" : "_" + fileNameAppend) + "." + tmpPath[1]);
            ImageIO.write((BufferedImage) itemp, "jpg", fileOut);

            /*try {
                if(mulFileServer == 1){
                    SFTPUtil.uploadFile1(fileOut,DateUtils.getShortDateStr(),fileOut.getName());
                }
            }catch (Exception e){
                e.printStackTrace();
            }*/

            return fileOut.getName();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return "";
    }

    public static void main(String[] args) throws IOException {
        //pressImage("C:\\Users\\nealmiao\\Desktop\\zg-20161230111332170.jpg", "C:\\Users\\nealmiao\\Desktop\\logo.png", 50, 40, 0.4f);
        //        pressText("C:\\Users\\nealmiao\\Desktop\\shl-20161230093135287.jpg", "问题描述………………", "宋体", Font.BOLD | Font.ITALIC, 20, Color.BLACK, 10, 10, 1f);
        //        resize("C:\\Users\\nealmiao\\Desktop\\shl-20161230101925918.jpg", 100, 500, true);
    }
}
